// $codepro.audit.disable
import static org.junit.Assert.*;

import org.junit.Test;
/**
 * JUnit test for the Doctor method EditTreatmentRecords
 * @author Team 9
 *
 */

public class DoctorEditTreatmentRecordsTest {
	/**
	 * The junit test for editting treatment records from the doctor class
	 */
	@Test
	public void testEditTreatmentRecords() {
		Patient p1 = new Patient("Patient", "One", "patient", "one");
		Patient p2 = new Patient("Patient", "Two", "patient", "two");
		Doctor d1 = new Doctor("Doctor", "One", "doctor", "one");
		Gui.addToPatientArray(p1);
		Gui.addToPatientArray(p2);
		d1.editTreatmentRecords("Patient", "One");
		d1.addTreatementRecord("Patient", "One");
		d1.editTreatmentRecords("Patient", "One");
		d1.editTreatmentRecords("Patient", "Three");
	}
}
